<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryRef" v-show="showSearch" :inline="true" label-width="68px">
      <el-form-item label="模型名称" prop="name">
        <el-input
            v-model="queryParams.name"
            placeholder="请输入名称"
            clearable
            style="width: 240px"
            @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item>
        <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
        <el-button icon="Refresh" @click="resetQuery">重置</el-button>
      </el-form-item>
    </el-form>
    <el-row :gutter="10" class="mb8">
      <el-col :span="1.5">
        <el-button
            type="primary"
            plain
            icon="Plus"
            @click="handleAdd"
            v-hasPermi="['ai:models:add']"
        >新增
        </el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
            type="success"
            plain
            icon="Edit"
            :disabled="single"
            @click="handleUpdate"
            v-hasPermi="['ai:models:update']"
        >修改
        </el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
            type="danger"
            plain
            icon="Delete"
            :disabled="multiple"
            @click="handleDelete"
            v-hasPermi="['ai:models:remove']"
        >删除
        </el-button>
      </el-col>
      <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <!-- 表格数据 -->
    <el-table v-loading="loading" :data="models" @selection-change="handleSelectionChange">
      <el-table-column type="selection" width="55" align="center"/>
      <el-table-column label="编号" prop="id" width="120"/>
      <el-table-column label="名称" prop="name" :show-overflow-tooltip="true" width="150"/>
      <el-table-column label="调用地址" prop="url" :show-overflow-tooltip="true" width="150"/>
      <el-table-column label="提问提示词" prop="questionPrompt" :show-overflow-tooltip="true" width="150"/>
      <el-table-column label="回答提示词" prop="answerPrompt" :show-overflow-tooltip="true" width="150"/>
      <el-table-column label="身份设定" prop="role" :show-overflow-tooltip="true" width="150"/>
      <el-table-column label="是否收费" prop="role" width="150">
        <template #default="scope">
          <el-tag type="primary" v-if="scope.row.charge==1">收费</el-tag>
          <el-tag type="success" v-else>免费</el-tag>
        </template>
      </el-table-column>
      <el-table-column label="是否支持多轮会话" prop="role" width="150">
        <template #default="scope">
          <el-tag type="primary" v-if="scope.row.multiple==1">是</el-tag>
          <el-tag type="success" v-else>否</el-tag>
        </template>
      </el-table-column>

      <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
        <template #default="scope">
          <el-tooltip content="修改" placement="top">
            <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)"
                       v-hasPermi="['ai:models:update']"></el-button>
          </el-tooltip>
          <el-tooltip content="删除" placement="top">
            <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)"
                       v-hasPermi="['ai:models:remove']"></el-button>
          </el-tooltip>
        </template>
      </el-table-column>
    </el-table>

    <pagination
        v-show="total > 0"
        :total="total"
        v-model:page="queryParams.pageNum"
        v-model:limit="queryParams.pageSize"
        @pagination="getList"
    />

    <!-- 添加或修改Banner配置对话框 -->
    <el-dialog :title="title" v-model="open" width="500px" append-to-body>
      <el-form ref="modelsRef" :model="form" :rules="rules" label-width="100px">
        <el-form-item label="模型名称" prop="name">
          <el-input v-model="form.name" placeholder="请输入模型名称"/>
        </el-form-item>
        <el-form-item label="调用地址" prop="url">
          <el-input v-model="form.url" placeholder="请输入调用地址"/>
        </el-form-item>
        <el-form-item label="AI身份设定" prop="role">
          <el-input v-model="form.role" placeholder="请输入AI身份设定"/>
        </el-form-item>
        <el-form-item label="提问提示词" prop="questionPrompt">
          <el-input v-model="form.questionPrompt" placeholder="请输入提问提示词"/>
        </el-form-item>
        <el-form-item label="回答提示词" prop="answerPrompt">
          <el-input v-model="form.answerPrompt" placeholder="请输入回答提示词"/>
        </el-form-item>
        <el-form-item label="排序字段" prop="orderNum">
          <el-input-number v-model="form.orderNum" controls-position="right" :min="0"/>
        </el-form-item>
        <el-form-item label="收费" prop="charge">
          <el-switch
              v-model="form.charge"
              style="--el-switch-on-color: #13ce66; --el-switch-off-color: #ff4949"
              active-value="1"
              active-text="是"
              inactive-text="否"
              inactive-value="0"/>
        </el-form-item>
        <el-form-item label="多轮会话" prop="multiple">
          <el-switch
              v-model="form.multiple"
              style="--el-switch-on-color: #1ba1dc; --el-switch-off-color: #ffc549"
              active-value="1"
              active-text="是"
              inactive-text="否"
              inactive-value="0"/>
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button type="primary" @click="submitForm">确 定</el-button>
          <el-button @click="cancel">取 消</el-button>
        </div>
      </template>
    </el-dialog>
  </div>
</template>

<script setup name="Role">

import {getToken} from "../../../utils/auth.js";
import {addCategory, delCategory, getCategory, listCategory, updateCategory} from "../../../api/ai/category.js";
import {listModels, addModels, updateModels, getModels,delModels} from "../../../api/ai/models.js";

const router = useRouter();
const {proxy} = getCurrentInstance();
const {sys_normal_disable} = proxy.useDict("sys_normal_disable");

const models = ref([]);
const open = ref(false);
const loading = ref(true);
const showSearch = ref(true);
const ids = ref([]);
const single = ref(true);
const multiple = ref(true);
const total = ref(0);
const title = ref("");
const dateRange = ref([]);
const headerObj = ref({
  'Authorization': 'Bearer ' + getToken()
})

const data = reactive({
  form: {},
  queryParams: {
    pageNum: 1,
    pageSize: 10,
    name: undefined,
    text: undefined
  },
  rules: {
    url: [{required: true, message: "模型地址不能为空", trigger: "blur"}],
    orderNum: [{required: true, message: "排序字段不能为空", trigger: "blur"}],
    charge: [{required: true, message: "收费字段不能为空", trigger: "blur"}],
    multiple: [{required: true, message: "多轮会话字段不能为空", trigger: "blur"}],
    role: [{required: true, message: "AI身份设定不能为空", trigger: "blur"}],
    answerPrompt: [{required: true, message: "回答提示词不能为空", trigger: "blur"}],
    questionPrompt: [{required: true, message: "提问提示词不能为空", trigger: "blur"}],
    name: [{required: true, message: "名称不能为空", trigger: "blur"}]
  },
});

const {queryParams, form, rules} = toRefs(data);

function fileUploadSuccess(res) {
  form.value.src = res.msg
}

/** 查询Banner列表 */
function getList() {
  loading.value = true;
  listModels(proxy.addDateRange(queryParams.value, dateRange.value)).then(response => {
    models.value = response.rows;
    total.value = response.total;
    loading.value = false;
  });
}

/** 搜索按钮操作 */
function handleQuery() {
  queryParams.value.pageNum = 1;
  getList();
}

/** 重置按钮操作 */
function resetQuery() {
  dateRange.value = [];
  proxy.resetForm("queryRef");
  handleQuery();
}

/** 删除按钮操作 */
function handleDelete(row) {
  const modelsIds = row.id || ids.value;
  proxy.$modal.confirm('是否确认删除编号为"' + modelsIds + '"的数据项?').then(function () {
    return delModels(modelsIds);
  }).then(() => {
    getList();
    proxy.$modal.msgSuccess("删除成功");
  }).catch(() => {
  });
}

/** 多选框选中数据 */
function handleSelectionChange(selection) {
  ids.value = selection.map(item => item.id);
  single.value = selection.length != 1;
  multiple.value = !selection.length;
}

/** 重置新增的表单以及其他数据  */
function reset() {
  form.value = {
    id: undefined,
    name: undefined,
    url: undefined,
    orderNum: undefined,
    charge: undefined,
    multiple: undefined,
    role: undefined,
    answerPrompt: undefined,
    questionPrompt: undefined
  };
  proxy.resetForm("modelsRef");
}

/** 添加Banner */
function handleAdd() {
  reset();
  open.value = true;
  title.value = "添加模型信息";
}

/** 修改Banner */
function handleUpdate(row) {
  reset();
  const id = row.id || ids.value;
  getModels(id).then(response => {
    form.value = response.data;
    open.value = true;
    title.value = "修改模型信息";
  });
}

function enableChange(data) {
  form.value = data;
  updateCategory(form.value).then(response => {
    proxy.$modal.msgSuccess("修改成功");
    open.value = false;
    getList();
  });
}

/** 提交按钮 */
function submitForm() {
  proxy.$refs["modelsRef"].validate(valid => {
    if (valid) {
      if (form.value.id != undefined) {
        updateModels(form.value).then(response => {
          proxy.$modal.msgSuccess("修改成功");
          open.value = false;
          getList();
        });
      } else {
        addModels(form.value).then(response => {
          proxy.$modal.msgSuccess("新增成功");
          open.value = false;
          getList();
        });
      }
    }
  });
}

/** 取消按钮 */
function cancel() {
  open.value = false;
  reset();
}

getList();
</script>
